package com.example.yibu.netty.manager;

import io.netty.channel.ChannelHandlerContext;
import java.util.HashMap;
import java.util.Map;

public class UserSessionManager {
    private final Map<String, ChannelHandlerContext> userSessions = new HashMap<>();

    public void addUser(String userId, ChannelHandlerContext ctx) {
        userSessions.put(userId, ctx);
    }

    public void removeUser(String userId) {
        userSessions.remove(userId);
    }

    public ChannelHandlerContext getUserSession(String userId) {
        return userSessions.get(userId);
    }
}
